Core Apps » ChatSecure:Android

//

//  OMEMODeviceVerificationViewController.swift

//  ChatSecure

//

//  Created by Chris Ballinger on 10/13/16.

//  Copyright © 2016 Chris Ballinger. All rights reserved.

//

import UIKit

import XLForm

import YapDatabase

import OTRAssets

open class UserProfileViewController: XLFormViewController {

    @objc open var completionBlock: (()->Void)?

    // Crypto Chooser row tags

    open static let DefaultRowTag = "DefaultRowTag"

    open static let PlaintextRowTag = "PlaintextRowTag"

    open static let OTRRowTag = "OTRRowTag"

    open static let OMEMORowTag = "OMEMORowTag"

    open static let ShowAdvancedCryptoSettingsTag = "ShowAdvancedCryptoSettingsTag"

    open let accountKey:String

    open var connection: YapDatabaseConnection

    lazy var signalCoordinator:OTROMEMOSignalCoordinator? = {

        var account:OTRAccount? = nil

        self.connection.read { (transaction) in

            account = OTRAccount.fetchObject(withUniqueID: self.accountKey, transaction: transaction)

        }

        guard let acct = account else {

            return nil

        }

        guard let xmpp = OTRProtocolManager.sharedInstance().protocol(for: acct) as? OTRXMPPManager else {

            return nil

        }

        return xmpp.omemoSignalCoordinator

    }()

    @objc public init(accountKey:String, connection: YapDatabaseConnection, form: XLFormDescriptor) {

        self.accountKey = accountKey

        self.connection = connection

        super.init(nibName: nil, bundle: nil)

        self.form = form

    }

    required public init!(coder aDecoder: NSCoder!) {

        fatalError("init(coder:) has not been implemented")

    }

    open override func viewDidLoad() {

        // gotta register cell before super

        OMEMODeviceFingerprintCell.registerCellClass(OMEMODeviceFingerprintCell.defaultRowDescriptorType())

        UserInfoProfileCell.registerCellClass(UserInfoProfileCell.defaultRowDescriptorType())

        super.viewDidLoad()

        self.navigationItem.rightBarButtonItem = UIBarButtonItem(barButtonSystemItem: .done, target: self, action: #selector(doneButtonPressed(_:)))

        self.tableView.allowsMultipleSelectionDuringEditing = false

        // Overriding superclass behaviour. This prevents the red icon on left of cell for deletion. Just want swipe to delete on device/fingerprint.

        self.tableView.setEditing(false, animated: false)

    }

    open override func didReceiveMemoryWarning() {

        super.didReceiveMemoryWarning()

        // Dispose of any resources that can be recreated.

    }

    @objc open func doneButtonPressed(_ sender: AnyObject?) {

        var devicesToSave: [OTROMEMODevice] = []

        var otrFingerprintsToSave: [OTRFingerprint] = []

        for (_, value) in form.formValues() {

            switch value {

            case let device as OTROMEMODevice:

                devicesToSave.append(device)

            case let fingerprint as OTRFingerprint:

                otrFingerprintsToSave.append(fingerprint)

            default:

                break

            }

        }

        OTRDatabaseManager.sharedInstance().readWriteDatabaseConnection?.asyncReadWrite({ (t: YapDatabaseReadWriteTransaction) in

            for viewedDevice in devicesToSave {

                if var device = t.object(forKey: viewedDevice.uniqueId, inCollection: OTROMEMODevice.collection) as? OTROMEMODevice {

                    device = device.copy() as! OTROMEMODevice

                    device.trustLevel = viewedDevice.trustLevel

                    if (device.trustLevel == .trustedUser && device.isExpired()) {

                        device.lastSeenDate = viewedDevice.lastSeenDate

                    }

                    device.save(with: t)

                }

            }

        })

        otrFingerprintsToSave.forEach { (fingerprint) in

            OTRProtocolManager.sharedInstance().encryptionManager.save(fingerprint)

        }

        if let completion = self.completionBlock {

            completion()

        }

        dismiss(animated: true, completion: nil)

    }

    fileprivate func isAbleToDeleteCellAtIndexPath(_ indexPath:IndexPath) -> Bool {

        if let rowDescriptor = self.form.formRow(atIndex: indexPath) {

            switch rowDescriptor.value {

            case let device as OTROMEMODevice:

                if let myBundle = self.signalCoordinator?.fetchMyBundle() {

                    // This is only used to compare so we don't allow delete UI on our device

                    let thisDeviceYapKey = OTROMEMODevice.yapKey(withDeviceId: NSNumber(value: myBundle.deviceId as UInt32), parentKey: self.accountKey, parentCollection: OTRAccount.collection)

                    if device.uniqueId != thisDeviceYapKey {

                        return true

                    }

                }

            case let fingerprint as OTRFingerprint:

                if (fingerprint.accountName != fingerprint.username) {

                    return true

                }

            default:

                break

            }

        }

        return false

    }

    fileprivate func performEdit(_ action:UITableViewCellEditingStyle, indexPath:IndexPath) {

        if ( action == .delete ) {

            if let rowDescriptor = self.form.formRow(atIndex: indexPath) {

                rowDescriptor.sectionDescriptor.removeFormRow(rowDescriptor)

                switch rowDescriptor.value {

                case let device as OTROMEMODevice:

                    self.signalCoordinator?.removeDevice([device], completion: { (success) in

                    })

                    break

                case let fingerprint as OTRFingerprint:

                    do {

                        try OTRProtocolManager.sharedInstance().encryptionManager.otrKit.delete(fingerprint)

                    } catch {

                    }

                    break

                default:

                    break

                }

            }

        }

    }

    open static func cryptoChooserRows(_ buddy: OTRBuddy, connection: YapDatabaseConnection) -> [XLFormRowDescriptor] {

        let bestAvailableRow = XLFormRowDescriptor(tag: DefaultRowTag, rowType: XLFormRowDescriptorTypeBooleanCheck, title: Best_Available())

        let plaintextOnlyRow = XLFormRowDescriptor(tag: PlaintextRowTag, rowType: XLFormRowDescriptorTypeBooleanCheck, title: Plaintext_Only())

        let plaintextOtrRow = XLFormRowDescriptor(tag: PlaintextRowTag, rowType: XLFormRowDescriptorTypeBooleanCheck, title: Plaintext_Opportunistic_OTR())

        let otrRow = XLFormRowDescriptor(tag: OTRRowTag, rowType: XLFormRowDescriptorTypeBooleanCheck, title: "OTR")

        let omemoRow = XLFormRowDescriptor(tag: OMEMORowTag, rowType: XLFormRowDescriptorTypeBooleanCheck, title: "OMEMO")

        var hasDevices = false

        connection.read { (transaction: YapDatabaseReadTransaction) in

            if OTROMEMODevice.allDevices(forParentKey: buddy.uniqueId, collection: type(of: buddy).collection, transaction: transaction).count > 0 {

                hasDevices = true

            }

        }

        if (!hasDevices) {

            omemoRow.disabled = NSNumber(value: true as Bool)

        }

        let trueValue = NSNumber(value: true as Bool)

        switch buddy.preferredSecurity {

        case .plaintextOnly:

            plaintextOnlyRow.value = trueValue

            break

        case .bestAvailable:

            bestAvailableRow.value = trueValue

            break

        case .OTR:

            otrRow.value = trueValue

            break

        case .OMEMO:

            omemoRow.value = trueValue

            break

        case .omemOandOTR:

            omemoRow.value = trueValue

            break

        case .plaintextWithOTR:

            plaintextOtrRow.value = trueValue

        }

        let formRows = [bestAvailableRow, plaintextOnlyRow, plaintextOtrRow, otrRow, omemoRow]

        var currentRow: XLFormRowDescriptor? = nil

        var rowsToDeselect: NSMutableSet = NSMutableSet()

        let onChangeBlock = { (oldValue: Any?, newValue: Any?, rowDescriptor: XLFormRowDescriptor) in

            // Prevent infinite loops

            // Allow deselection

            if rowsToDeselect.count > 0 {

                rowsToDeselect.remove(rowDescriptor)

                return

            }

            if currentRow != nil {

                return

            }

            currentRow = rowDescriptor

            // Don't allow user to unselect a true value

            if (newValue as AnyObject?)?.boolValue == false {

                rowDescriptor.value = NSNumber(value: true as Bool)

                currentRow = nil

                return

            }

            // Deselect other rows

            rowsToDeselect = NSMutableSet(array: formRows.filter({ $0 != rowDescriptor }))

            for row in rowsToDeselect {

                guard let row = row as? XLFormRowDescriptor else {

                    continue

                }

                let newValue = NSNumber(value: false as Bool)

                row.value = newValue

                // Wow that's janky

                (row.sectionDescriptor.formDescriptor.delegate as! XLFormViewControllerDelegate).reloadFormRow!(row)

            }

            var preferredSecurity: OTRSessionSecurity = .bestAvailable

            if (plaintextOnlyRow.value as AnyObject?)?.boolValue == true {

                preferredSecurity = .plaintextOnly

            } else if (otrRow.value as AnyObject?)?.boolValue == true {

                preferredSecurity = .OTR

            } else if (omemoRow.value as AnyObject?)?.boolValue == true {

                preferredSecurity = .OMEMO

            } else if (bestAvailableRow.value as AnyObject?)?.boolValue == true {

                preferredSecurity = .bestAvailable

            } else if (plaintextOtrRow.value as AnyObject?)?.boolValue == true {

                preferredSecurity = .plaintextWithOTR

            }

            OTRDatabaseManager.sharedInstance().readWriteDatabaseConnection?.readWrite({ (transaction: YapDatabaseReadWriteTransaction) in

                guard var buddy = transaction.object(forKey: buddy.uniqueId, inCollection: type(of: buddy).collection) as? OTRBuddy else {

                    return

                }

                guard let account = buddy.account(with: transaction) else {

                    return

                }

                buddy = buddy.copy() as! OTRBuddy

                buddy.preferredSecurity = preferredSecurity

                buddy.save(with: transaction)

                // Cancel OTR session if plaintext or omemo only

                if (preferredSecurity == .plaintextOnly || preferredSecurity == .OMEMO) {

                    OTRProtocolManager.sharedInstance().encryptionManager.otrKit.disableEncryption(withUsername: buddy.username, accountName: account.username, protocol: account.protocolTypeString())

                }

            })

            currentRow = nil

        }

        for row in formRows {

            row.onChangeBlock = onChangeBlock

        }

        return formRows

    }

//MARK UITableView Delegate overrides

    open override func tableView(_ tableView: UITableView, canEditRowAt indexPath: IndexPath) -> Bool {

        if  self.isAbleToDeleteCellAtIndexPath(indexPath) {

            return true

        }

        return false

    }

    open override func tableView(_ tableView: UITableView, editingStyleForRowAt indexPath: IndexPath) -> UITableViewCellEditingStyle {

        if  self.isAbleToDeleteCellAtIndexPath(indexPath) {

            return .delete

        }

        return .none

    }

    open override func tableView(_ tableView: UITableView, commit editingStyle: UITableViewCellEditingStyle, forRowAt indexPath: IndexPath) {

        self.performEdit(editingStyle, indexPath: indexPath)

    }

    @objc open static func profileFormDescriptorForAccount(_ account: OTRAccount, buddies: [OTRBuddy], connection: YapDatabaseConnection) -> XLFormDescriptor {

        let form = XLFormDescriptor(title: Profile_String())

        let yourProfileSection = XLFormSectionDescriptor.formSection(withTitle: Me_String())

        let yourProfileRow = XLFormRowDescriptor(tag: account.uniqueId, rowType: UserInfoProfileCell.defaultRowDescriptorType())

        yourProfileRow.value = account

        yourProfileSection.addFormRow(yourProfileRow)

        guard let xmpp = OTRProtocolManager.sharedInstance().protocol(for: account) as? OTRXMPPManager else {

            return form

        }

        guard let myBundle = xmpp.omemoSignalCoordinator?.fetchMyBundle() else {

            return form

        }

        let thisDevice = OTROMEMODevice(deviceId: NSNumber(value: myBundle.deviceId as UInt32), trustLevel: .trustedUser, parentKey: account.uniqueId, parentCollection: type(of: account).collection, publicIdentityKeyData: myBundle.identityKey, lastSeenDate: Date())

        var ourDevices: [OTROMEMODevice] = []

        connection.read { (transaction: YapDatabaseReadTransaction) in

            ourDevices = OTROMEMODevice.allDevices(forParentKey: account.uniqueId, collection: type(of: account).collection, transaction: transaction)

        }

        let ourFilteredDevices = ourDevices.filter({ (device: OTROMEMODevice) -> Bool in

            return device.uniqueId != thisDevice.uniqueId

        })

        // TODO - Sort ourDevices and theirDevices by lastSeen

        let addDevicesToSection: ([OTROMEMODevice], XLFormSectionDescriptor) -> Void = { devices, section in

            for device in devices {

                guard let _ = device.publicIdentityKeyData else {

                    continue

                }

                let row = XLFormRowDescriptor(tag: device.uniqueId, rowType: OMEMODeviceFingerprintCell.defaultRowDescriptorType())

                row.value = device.copy()

                // Don't allow editing of your own device

                if device.uniqueId == thisDevice.uniqueId {

                    row.disabled = true

                }

                section.addFormRow(row)

            }

        }

        let otrKit = OTRProtocolManager.sharedInstance().encryptionManager.otrKit

        let allFingerprints = otrKit.allFingerprints()

        let myFingerprint = otrKit.fingerprint(forAccountName: account.username, protocol: account.protocolTypeString())

        let addFingerprintsToSection: ([OTRFingerprint], XLFormSectionDescriptor) -> Void = { fingerprints, section in

            for fingerprint in fingerprints {

                let row = XLFormRowDescriptor(tag: (fingerprint.fingerprint as NSData).otr_hexString(), rowType: OMEMODeviceFingerprintCell.defaultRowDescriptorType())

                if let myFingerprint = myFingerprint {

                    if (fingerprint === myFingerprint) {

                        // We implicitly trust ourselves with OTR

                        row.disabled = true

                    } else {

                        row.disabled = false

                    }

                }

                row.value = fingerprint

                section.addFormRow(row)

            }

        }

        var allMyDevices: [OTROMEMODevice] = []

        allMyDevices.append(thisDevice)

        allMyDevices.append(contentsOf: ourFilteredDevices)

        addDevicesToSection(allMyDevices, yourProfileSection)

        var theirSections: [XLFormSectionDescriptor] = []

        if let myFingerprint = myFingerprint {

            addFingerprintsToSection([myFingerprint], yourProfileSection)

        }

        // Add section for each buddy's device

        for buddy in buddies {

            let theirSection = XLFormSectionDescriptor.formSection(withTitle: buddy.username)

            let buddyRow = XLFormRowDescriptor(tag: buddy.uniqueId, rowType: UserInfoProfileCell.defaultRowDescriptorType())

            buddyRow.value = buddy

            theirSection.addFormRow(buddyRow)

            var theirDevices: [OTROMEMODevice] = []

            connection.read({ (transaction: YapDatabaseReadTransaction) in

                theirDevices = OTROMEMODevice.allDevices(forParentKey: buddy.uniqueId, collection: type(of: buddy).collection, transaction: transaction)

            })

            let theirFingerprints = allFingerprints.filter({ (fingerprint: OTRFingerprint) -> Bool in

                return fingerprint.username == buddy.username &&

                fingerprint.accountName == account.username

            })

            addDevicesToSection(theirDevices, theirSection)

            addFingerprintsToSection(theirFingerprints, theirSection)

            theirSections.append(theirSection)

        }

        var sectionsToAdd: [XLFormSectionDescriptor] = []

        sectionsToAdd.append(contentsOf: theirSections)

        // cryptoChooserRows is only meaningful for 1:1 conversations at the moment

        if buddies.count == 1 {

            let buddy = buddies.first!

            let cryptoSection = XLFormSectionDescriptor.formSection(withTitle: Advanced_Encryption_Settings())

            cryptoSection.footerTitle = Advanced_Crypto_Warning()

            let showAdvancedSwitch = XLFormRowDescriptor.init(tag: self.ShowAdvancedCryptoSettingsTag, rowType: XLFormRowDescriptorTypeBooleanSwitch, title: Show_Advanced_Encryption_Settings())

            showAdvancedSwitch.value = NSNumber(value: false as Bool)

            let cryptoChooser = cryptoChooserRows(buddy, connection: connection)

            for row in cryptoChooser {

                cryptoSection.addFormRow(row)

            }

            cryptoSection.hidden = "$\(ShowAdvancedCryptoSettingsTag)==0"

            let buddySection = theirSections.first!

            buddySection.addFormRow(showAdvancedSwitch)

            sectionsToAdd.append(cryptoSection)

        }

        sectionsToAdd.append(yourProfileSection)

        for section in sectionsToAdd {

            if section.formRows.count > 0 {

                form.addFormSection(section)

            }

        }

        return form

    }

    // MARK: - UITableViewDelegate

    open override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {

        super.tableView(tableView, didSelectRowAt: indexPath)

        tableView.deselectRow(at: indexPath, animated: true)

        guard let cell = self.tableView(tableView, cellForRowAt: indexPath) as? OMEMODeviceFingerprintCell else {

            return

        }

        var fingerprint = ""

        var username = ""

        var cryptoType = ""

        if let device = cell.rowDescriptor.value as? OTROMEMODevice {

            cryptoType = "OMEMO"

            fingerprint = device.humanReadableFingerprint

            self.connection.read({ (transaction) in

                if let buddy = transaction.object(forKey: device.parentKey, inCollection: device.parentCollection) as? OTRBuddy {

                    username = buddy.username

                }

            })

        }

        if let otrFingerprint = cell.rowDescriptor.value as? OTRFingerprint {

            cryptoType = "OTR"

            fingerprint = (otrFingerprint.fingerprint as NSData).humanReadableFingerprint()

            username = otrFingerprint.username

        }

        if fingerprint.count == 0 || username.count == 0 || cryptoType.count == 0 {

            return

        }

        let stringToShare = "\(username): \(cryptoType) \(fingerprint)"

        let activityViewController = UIActivityViewController(activityItems: [stringToShare], applicationActivities: nil)

        if let ppc = activityViewController.popoverPresentationController {

            ppc.sourceView = cell

            ppc.sourceRect = cell.frame

        }

        present(activityViewController, animated: true, completion: nil)

    }

}